import { Request, Response } from 'express';
import axios from 'axios';

// 设置API Key
const DOCMEE_API_KEY = 'ak_s41kObF3rTp5TEAvtT';

interface DocmeeResponse {
  code: number;
  data?: {
    token?: string;
  };
  message?: string;
}

export const getDocmeeToken = async (req: Request, res: Response) => {
  try {
    const response = await axios.post<DocmeeResponse>('https://docmee.cn/api/user/createApiToken', {
      uid: 'user_' + Date.now(), // 用户唯一标识，可以替换为您系统的用户ID
      limit: null // 不限制生成PPT次数，如果您希望限制，可以设置为具体数字
    }, {
      headers: {
        'Content-Type': 'application/json',
        'Api-Key': DOCMEE_API_KEY
      }
    });

    if (response.data && response.data.code === 0 && response.data.data && response.data.data.token) {
      res.json({
        code: 0,
        data: {
          token: response.data.data.token
        }
      });
    } else {
      res.status(500).json({
        code: 1,
        message: '获取token失败: ' + (response.data?.message || '未知错误')
      });
    }
  } catch (error: any) {
    console.error('获取Docmee token失败:', error);
    res.status(500).json({
      code: 1,
      message: '获取token失败: ' + (error.response?.data?.message || error.message || '未知错误')
    });
  }
}; 